import torch
# 加载权重文件
checkpoint = torch.load('./model_out/Synapse/SwinUnet0/best_model.pth')

# 打印权重文件中的键
print("权重文件中的键：")
print(checkpoint.keys())

# 检查输出层的权重形状
if 'swin_unet.output.weight' in checkpoint:
    print("输出层权重形状：", checkpoint['swin_unet.output.weight'].shape)
else:
    print("未找到输出层权重")